import os
import shutil

'''
如果 路径A 中存在 a.jpg 但没有 a.txt，就把 a.jpg 移动到 路径B
如果 路径A 中存在 b.txt 但没有 b.jpg，就把 b.txt 移动到 路径B
'''

# 配置路径
path_A = r"E:\rune\us_yolo"  # 原始数据集路径（替换为你的实际路径）
path_B = r"E:\rune\us_yolo--"  # 目标存放路径（替换为你的实际路径）
os.makedirs(path_B, exist_ok=True)
# 获取路径A中的所有文件
files = set(os.listdir(path_A))

# 遍历文件
for file in files:
    file_path = os.path.join(path_A, file)

    if file.endswith(".jpg"):
        txt_file = file.replace(".jpg", ".txt")
        if txt_file not in files:
            shutil.move(file_path, os.path.join(path_B, file))
            print(f"Moved {file} to {path_B}")

    elif file.endswith(".txt"):
        jpg_file = file.replace(".txt", ".jpg")
        if jpg_file not in files:
            shutil.move(file_path, os.path.join(path_B, file))
            print(f"Moved {file} to {path_B}")